package tree;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层，而根节点的子节点位于第 2 层，依此类推。
 请你找出层内元素之和 最大 的那几层（可能只有一层）的层号，并返回其中 最小 的那个。
 输入：root = [1,7,0,7,-8,null,null]
 输出：2
 解释：
 第 1 层各元素之和为 1，
 第 2 层各元素之和为 7 + 0 = 7，
 第 3 层各元素之和为 7 + -8 = -1，
 所以我们返回第 2 层的层号，它的层内元素之和最大。
 * @author le
 */
public class MaxLevelSumOfBinaryTree
{
    public static void main(String[] args)
    {

    }

    public int maxLevelSum(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        if (root != null){
            queue.offer(root);
        }
        List<Integer> list = new ArrayList<>();
        while(!queue.isEmpty()){
            int size = queue.size();
            int sum = 0;
            while(size-- > 0){
                TreeNode node = queue.poll();
                sum += node.val;
                if (node.left != null){
                    queue.offer(node.left);
                }
                if (node.right != null){
                    queue.offer(node.right);
                }
            }
            list.add(sum);
        }
        int index = 0;
        int min = Integer.MAX_VALUE;
        for (int i = 0;i < list.size();i++){
            if (min > list.get(i)){
                min = list.get(i);
                index = i;
            }
        }
        return index;
    }
}
